package cn.xttc.edu.controller;

import cn.xttc.edu.entity.CarType;
import cn.xttc.edu.entity.ReturnMsg;
import cn.xttc.edu.entity.vo.CarTypeVo;
import cn.xttc.edu.service.CarTypeService;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.util.ArrayList;
import java.util.List;

/**
 * <p>
 *  前端控制器
 * </p>
 *
 * @author 杨磊
 * @since 2022-03-09
 */
@RestController
public class CarTypeController {

    @Autowired
    CarTypeService carTypeService;

    /**
     * 根据id查询类型
     * @param id id
     * @return 包装对象
     */
    @GetMapping("/carType/{id}")
    public ReturnMsg getById(@PathVariable("id") int id) {
        CarType carType = carTypeService.getById(id);
        return ReturnMsg.success(carType);
    }

    /**
     * 获取所有汽车类型
     * @return 包装对象
     */
    @GetMapping("/carType")
    public ReturnMsg getAll() {
        List<CarType> list = carTypeService.list();
        if (!list.isEmpty()) {
            return ReturnMsg.success(list);
        } else {
            return ReturnMsg.fail(null);
        }
    }

    /**
     * 分页获取货车类型
     * @param pn 页码
     * @return 包装对象
     */
    @GetMapping("/pageCarType")
    public ReturnMsg pageGetAll(int pn) {
        Page<CarTypeVo> page = new Page<>(pn, 5);
        carTypeService.pageGetAll(page);
        if (!page.getRecords().isEmpty()) {
            return ReturnMsg.success(page);
        } else {
            return ReturnMsg.fail("");
        }
    }

    /**
     * 新增货车类型
//     * @param carType 类型
     * @return 包装对象
     */
    @PostMapping("/carType")
    public ReturnMsg addCarType(CarType carType) {
        try {
            carTypeService.save(carType);
            return ReturnMsg.success("添加成功");
        } catch (Exception e) {
            return ReturnMsg.fail("添加失败，请联系管理员");
        }
    }

    /**
     * 更新货车类型
     * @param carType 类型数据
     * @return 包装对象
     */
    @PutMapping("/carType/{id}")
    public ReturnMsg editCarType(CarType carType) {
        boolean b = carTypeService.updateById(carType);
        if (b) {
            return ReturnMsg.success("修改成功");
        } else {
            return ReturnMsg.fail("修改失败");
        }
    }

    /**
     * 单个删除和批量删除
     * @param ids 多个id拼接的字符串
     * @return 包装对象
     */
    @DeleteMapping("/carType/{ids}")
    public ReturnMsg delete(@PathVariable("ids") String ids) {
        if (ids.contains("-")) {
            //批量删除
            List<Integer> delIds = new ArrayList<>();
            String[] strIds = ids.split("-");
            //组装id的集合
            for (String string : strIds) {
                delIds.add(Integer.parseInt(string));
            }
            try {
                carTypeService.removeBatchByIds(delIds);
            } catch (Exception e) {
                return ReturnMsg.fail("删除失败，该货车类型已与其他内容绑定");
            }
        } else {
            //单个删除
            Integer id = Integer.parseInt(ids);
            try {
                carTypeService.removeById(id);
            } catch (Exception e) {
                return ReturnMsg.fail("删除失败，该货车类型已与其他内容绑定");
            }
        }
        return ReturnMsg.success("删除成功");
    }

}

